from selenium.webdriver.support.ui import WebDriverWait
import time

class Base:
    def __init__(self,driver):
        self.driver = driver
    #打开网址
    def open(self,url):
        self.driver.get(url)
    #定位元素
    def find_element(self,loc):
        try:
            WebDriverWait(self.driver,20,1).until(lambda driver:self.driver.find_element(*loc).is_displayed())
            return self.driver.find_element(*loc)
        except Exception as e:
           t = time.strftime("%Y_%m_%d_%H_%M_%S",time.localtime())
           self.driver.save_screenshot(t+".png")
           return e
    #点击
    def click(self,loc):
        self.find_element(loc).click()
    #输入
    def send_keys(self,loc,info):
        self.find_element(loc).send_keys(info)
    #文本
    def text(self,loc):
       return self.find_element(loc).text

    #截图
    def jietu(self):
        self.driver.save_screenshot(".png")
    #切窗口
    def qiehuan(self):
        self.driver.switch_to.window(self.driver.window_handles[-1])